import unittest
from employ import Employee

class EmployeeTestCase(unittest.TestCase):
	def setUp(self):
		self.employee = Employee("zhang", "san", 3000)

	def test_give_default_raise(self):
		salary = self.employee.give_raise()
		self.assertEqual(salary, 8000)

	def test_give_custom_raise(self):
		salary = self.employee.give_raise(3000)
		self.assertEqual(salary, 6000)
		
if __name__ == '__main__':
	unittest.main()